Skip to main content

MUSA SDK v5.2.0 发布版本信息

下载链接

发布说明

MUSA SDK v5.2.0 围绕 MUSA Driver/Runtime、编译工具链、核心数学库、深度学习库、通信库和调试分析工具进行了能力扩展与稳定性提升,重点增强通用计算兼容能力、Graph 能力和编译器能力,升级 Triton 版本支持,扩展 Math-X API 覆盖,并完善 MCCL 通信能力与多平台部署验证。

本版本已完成多平台发布测试。测试结论显示,S5000 在 Intel、Hygon4、AMD 等 CPU 平台及 Alinux、VesselOS、Ubuntu、Kylin、Bclinux 等系统环境下功能可用,模型性能达标。部分 CTS 或专项测试存在已知问题,评估为不影响模型性能或不影响本次版本发布,作为 Open Issue 后续跟踪。

具体支持平台详见下方列表。

重点更新内容

  1. MUSA Graph 功能增强,新增 33 个 Graph API。
  2. 编译器功能增强,新增 46 个 FP128 相关 API 支持。
  3. Triton 生态支持增强,升级 Triton 兼容版本至 3.6.0。
  4. muDNN 新增 5 个 C 接口 API。
  5. 数学库功能增强,新增 17 个 muBLAS / muBLASLt API 支持,新增 8 个 muSOLVER API 支持。
  6. MCCL 支持 GPU 对等内存直接访问(DevComm / Window / LSA Pointer / Team-LSA),GPU 可直接读写其他 GPU 数据,通信可在 kernel 内部发起,降低 MoE 专家并行、投机解码等场景的通信延迟。

支持平台

S5000 支持平台

MUSA SDK 支持 x86 架构 CPU,并通过 DEB 和 RPM 包覆盖多种 OS 组合。下述环境为推荐使用环境。

CPUOS内核版本MUSA SDK 包
IntelAlibaba Cloud Linux 3 (Soaring Falcon)5.10.134-13.al8.x86_64RPM
IntelVesselOS 2.0 (LTS-SP2)6.6.0-100.jd_b007.x86_64RPM
IntelUbuntu 24.04.1 LTS6.8.0-124-genericDEB
IntelUbuntu 22.04.4 LTS5.15.0-105-genericDEB
Hygon4Ubuntu 22.04.4 LTS5.15.0-105-genericDEB
AMDUbuntu 22.04.4 LTS5.15.0-105-genericDEB
Hygon4Kylin Linux Advanced Server V10 (Halberd)4.19.90-89.11.v2401.ky10.x86_64RPM
Hygon4Kylin Linux Advanced Server V11 (Swan25)6.6.0-32.7.v2505.ky11.x86_64RPM
Hygon4BigCloud Enterprise Linux For Euler 21.10 LTS4.19.90-2107.6.0.0208.16.oe1.bclinux.x86_64RPM

产品说明

摩尔线程 MUSA SDK 是一套完整的 GPU 并行计算开发环境,专为利用摩尔线程 GPU 加速程序而设计。通过 MUSA SDK,用户可以基于摩尔线程 GPU 构建和部署高性能计算、人工智能训练与推理、科学计算、通信与多媒体等应用。

MUSA SDK 包含 GPU 加速库、调试和优化工具、C/C++ 编译器以及用于应用程序部署的运行时库等功能。作为一套完整的开发工具,MUSA SDK 包含以下主要组件:

  • MUSA Toolkits:包含底层编译器、MUSA 运行时库、Musify 工具以及 MUSA-X 基础数学计算加速库。
  • muDNN:MT GPU 深度学习加速库。
  • MCCL:MT GPU 通信加速库。
  • Triton-MUSA:支持在摩尔线程 GPU 上运行 Triton DSL。
  • TileLang-MUSA:支持在摩尔线程 GPU 上运行 TileLang 相关工作负载。
  • MATE:算子加速库。
  • MUSA 工具链:包含 MUPTI、msys、mcu、MUSA Snapshot、MUSA sanitizer、MUSA gdb 等调试、分析与诊断工具。
  • MUSA SDK 依赖摩尔线程 GPU 和摩尔线程 GPU 通用驱动程序,需要运行在带有摩尔线程 GPU 的服务器、工作站或 PC 中,并安装对应版本的通用 Linux 驱动程序。

功能描述

  • 支持 SIMT (Single Instruction Multiple Thread) 架构并行编程模型。
  • 提供 MUSA 编程语言配套编译器工具链 mcc,用于将 MUSA 应用程序源码编译为可在 GPU 上执行的程序。
  • 提供 MUSA Driver/Runtime,支撑基于 MT GPU 的完整运行环境。
  • 提供 MUSA 数学库,包括 muBLAS、muRAND、muFFT、muSPARSE、muSOLVER、muPP、muThrust、muAlg 等。
  • 提供 muDNN 深度学习加速库,支撑神经网络算子和模型训练、推理场景。
  • 提供 MCCL 通信库,支持多 GPU、多节点通信原语和性能优化。
  • 提供 Triton-MUSA 和 TileLang-MUSA,支撑大模型训练、推理和算子开发生态。
  • 提供 MUPTI、msys、mcu、MUSA Snapshot、MUSA sanitizer 等调试、性能分析和诊断能力。

主要更新

功能更新(Feature Updates)

MUSA Driver/Runtime

  • 支持 pfm dump 在 UserQ 模式下工作。
  • MUPTI 支持 graph 粒度的 user queue graph trace。
  • MUSA sanitizer 增强 spill、printf、VMM API 相关定位能力。
  • 本版本验证 33 个 Graph API。
  • 增强 MUSA Snapshot 功能,提升问题定位和可调试性。
  • 下列能力以实验特性提供,后续版本将逐步转为正式发布能力:
    • Error Log Management Functions
    • Graph Management API support
    • MTSHMEM DeepEP normal / ll_compatibility 相关能力

mtcc 编译工具链

  • use_fast_math 编译选项对齐 NVCC 语义。
  • 支持 MTRTC。
  • 新增 46 个 FP128 相关 API 支持。
  • 支持 -dc 场景下的 MUSA automatic RDC。

Triton-MUSA

  • 支持 Triton 3.2.0。
  • 支持 Triton 3.6.0。

Math-X

  • muBLAS / muBLASLt 新增 17 个 API 支持,增强 emulation 相关 API 能力,持续优化 VASP、HPC 等场景表现。
  • muSOLVER 新增 8 个 API 支持。

muDNN

  • 新增 mudnnGathermudnnUnfoldmudnnScanmudnnWeightNormmudnnL2Loss 等 5 个 backend C API 支持。

MCCL

  • 支持 MCCL Base Comm / CommProperties。
  • 支持 MCCL DevComm / Window / LSA Pointer / Team-LSA。

Linux Driver

  • 支持 EDC counter 相关变更。

已知问题与限制

mtbios > 4.3.43 时,EDC counter 相关变更需要 Linux Driver 5.2.0、dcgm 1.1.6 和 mtml 2.4.2 配合使用。

产品组件版本说明

组件版本
MT Linux Driver5.2.0
MUSA Toolkit5.2.0
muDNN3.4.0
MCCL2.4.0
Triton-MUSA3.6.0
TileLang-MUSA0.1.8
MATE0.2.3